


np.where(a==b)  #返回满足条件的张量位置索引

